Patch Embedding Pytorch Vision Transformer